import java.util.HashMap;

class Solution2 {
    public int subarraysDivByK(int[] nums, int k) {
        int sum = 0, ret = 0,x = 0;
        HashMap<Integer,Integer> map = new HashMap<>();
        map.put(0,1);
        for(int i = 0; i < nums.length; i++) {
            sum += nums[i];
            int r = (sum % k + k) % k;
            ret += map.getOrDefault(r,0);
            map.put(r,map.getOrDefault(r,0) + 1);
        }
        return ret;
    }
}